import matplotlib
import xlrd
import matplotlib.pyplot as plt
matplotlib.use('TkAgg')

data=xlrd.open_workbook('data/1.xls')
table=data.sheets()[0]
sen=table.col_values(2)#读表格的第三列（情绪值）
print(sen)
s=input('(0直方图；1饼图；)请输入要绘制的统计图序号：')
if s=='0':#绘制直方图

    plt.hist(sen, bins=10, range=(0, 1))
    plt.title('Emotional value distribution')
    plt.show()
if s=='1':#绘制饼图
    sum1=0
    sum2=0
    sum3=0
    sum4=0
    for each in sen:#统计不同区间的情绪值分布
        if 0<each<0.25:
            sum1+=1
        if 0.25<each<0.5:
            sum2+=1
        if 0.5<each<0.75:
            sum3+=1
        if 0.75<each<1:
            sum4+=1
    x=[sum1,sum2,sum3,sum4]
    labels=['0~0.25','0.25~0.5','0.5~0.75','0.75~1']
    plt.pie(x,labels=labels)
    plt.title('Emotional value distribution')
    plt.show()